/*
 * Copyright (c) 2008, University of Aarhus.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * - Redistributions of source code must retain the above copyright notice,
 *   this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - Neither the name of the University of Aarhus nor the names
 *   of its contributors may be used to endorse or promote products derived
 *   from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
 * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
 * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
 * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * @author: Morten Tranberg Hansen <mth@cs.au.dk>
 * @date:   October 16 2008	
 * 
 */

package dk.sensobyg.network.tinyos;

import java.io.IOException;

import net.tinyos.message.Message;
import net.tinyos.message.MessageListener;
import net.tinyos.message.MoteIF;
import net.tinyos.packet.BuildSource;
import net.tinyos.packet.PhoenixError;
import net.tinyos.packet.PhoenixSource;
import net.tinyos.util.PrintStreamMessenger;

import org.apache.log4j.Logger;

import dk.sensobyg.base.*;

public class TinyOSWrapper extends AbstractWrapper implements MessageListener {

	private transient static Logger log = Logger.getLogger("dk.sensobyg.network.tinyos.TinyOSWrapper");
	
	private SafeStorage<SerialDataMessage> storage;
	private MoteIF mote;
	private boolean running;
	
	public TinyOSWrapper(Collector collector) {
		super(collector);
		storage = new SimpleSafeStorage<SerialDataMessage>(); //TODO: Use proper safe storage implementation.
		mote = null;
		running = false;
	}

	@Override
	public void run() {
		
		/*log.debug("Name: " + getName());
		log.debug("Description: " + getDescription());
		log.debug("URI: " + getURI());
		for(AbstractMote mote : getMotes()) {
			String r = "Mote " + mote.getMoteId() + ": ";
			for(AbstractMote.Sensor s : mote.getSensors()) {
				r += "[id=" + s.getSensorId() + "; type=" + s.getType() + "] "; 
			}
			log.debug(r);
		}*/
		
		PhoenixSource source = BuildSource.makePhoenix(getURI().getUserInfo()+"@"+getURI().getHost()+":"+getURI().getPort(), PrintStreamMessenger.err);
		if(source==null) {
			log.error("URI " + getURI() + " is not a valid tinyos URI.");
			return;
		}
		source.setPacketErrorHandler(new PhoenixError() {
			public void error(IOException e) {
			    try {
			    	log.error("TinyOSWrapper " + getName() + " could not connect to " + getURI() + ", retrying in 2 seconds.");
			    	sleep(2000);
			    }
			    catch (InterruptedException ie) { }
			}
		    });
		source.setPriority(Thread.MAX_PRIORITY);
		
		mote = new MoteIF(source);		
		mote.registerListener(new SerialDataMessage(), this);
		
		running = true;
		while(running) {
			try {
				SerialDataMessage message = storage.take();
				TinyOSMote mote = getMote(message.get_origin());
				
				if(mote!=null) {
					for(SensorData data : mote.getData(message)) {
						collector.newData(data);
					}
				} else {
					log.error("Received data from an unknown mote.");
				}
				
			} catch (InterruptedException e) {
				running = false;
			}
		}

		mote.deregisterListener(new SerialDataMessage(), this);
		source.shutdown();
		
		log.debug("Closing down wrapper.");
	}
	
	private TinyOSMote getMote(int moteid) {
		for(AbstractMote mote : getMotes()) {
			if(mote instanceof TinyOSMote) {
				TinyOSMote tmote = (TinyOSMote) mote;
				if(tmote.getMoteId()==moteid) {
					return tmote;
				}
			}			
		}
		return null;
	}
	
	@Override
	public void configure(Configuration configuration) {
		if(running) {
			CommandMessage message = null;
			
			if(configuration.getType()==ConfigurationType.SAMPLE_INTERVAL) {
				message = new CommandMessage();
				message.set_type(Types.COMMAND_TYPE_TEMPERATURE);
				message.set_interval(configuration.getInterval());
				message.set_target(Constants.BROADCAST_ADDR);
				message.amTypeSet(Constants.AM_SENSOBYG_COMMAND_MSG);
	    	}
			//TODO: Implement more commands.
			
			if(message!=null) {
				try {
					mote.send(0/*MoteIF.TOS_BCAST_ADDR*/, message);
				}
				catch (IOException e) {
					log.error("Cannot send configuration message to mote.", e);
				}
			}
		}
	}
	
	public void messageReceived(int addr, Message msg) {
		if(msg instanceof SerialDataMessage) {
			SerialDataMessage serial = (SerialDataMessage) msg;
			try {
				storage.put(serial);
			} catch (InterruptedException e) {
				log.error("SerialDataMessage lost.");
			}
		}
	}
}
